package com.tanhua.autoconfig.template;

import com.baidu.aip.face.AipFace;
import org.json.JSONObject;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.HashMap;

/**
 * @Author: JW
 * @Description: TODO 这是人脸识别
 * @DateTime: 2021/12/21 20:15
 **/
public class AipTemplate {

    @Autowired
    private AipFace aipFace;

    /**
     * @Author: JW
     * @Description:  TODO 这是人脸识别
     * @Date: 20:32 2021/12/21
     * @Param: [url] 图片路径
     * @return: boolean true为真人图片
    **/
    public boolean checkFace(String url){

        //选择图片路径
        String imageType = "URL";

        // 传入可选参数调用接口
        HashMap<String, String> options = new HashMap<String, String>();
        options.put("face_field", "age");
        options.put("max_face_num", "1");

        // 人脸检测
        JSONObject res = aipFace.detect(url, imageType,options);

        int error_code = (int) res.get("error_code");

        if (error_code == 0){
            return true;
        }

        return false;
    }
}
